package com.dyz.leetcode._18_backtracking;

import java.util.ArrayList;
import java.util.List;



public class _51_solveNQueens {
    //定义一个max表示共有多少个皇后
    int max;
    //保存8皇后的结果
    int[] arr;
    static int judgecount =0;
    /**
     * 放置第n个皇后
     */
    public void check(int n){
        if (n == max){
            print();
            return;
        }
        //一次放入皇后 ，在判断是否冲突
        for (int i = 0; i < max; i++) {
            //先把当前皇后 n 放到该行的第1列
            arr[n] = i;
            //判断放置第n个皇后到i列时，是否冲突
            if (judge(n)){
                //不冲突
                check(n+1);
            }
            //如果冲突，就继续 arr[n] = i;，即将第n个皇后，放置本行的后移一个位置

        }
    }

    /**
     * 判断该皇后与已经摆放的皇后是否冲突
     */
    private boolean judge(int n){
        judgecount++;
        for (int i = 0; i < n; i++) {
            //arr[i] == arr[n]  判断第n个皇后是否和前面n-1皇后是否在同一列
            //Math.abs(n-i) == Math.abs(arr[n]-arr[i]) 判断第n个皇后是否和第i皇后是否在同一斜线
            if (arr[i] == arr[n] || Math.abs(n-i) == Math.abs(arr[n]-arr[i])){
                return false;
            }
        }
        return true;
    }

    /**
     * 打印结果
     */
    public void print(){
        for (int i = 0; i < arr.length; i++) {
            System.out.printf("%d ",arr[i]);
        }
        System.out.printf("\n");
    }

    public List<List<String>> solveNQueens(int n) {
        List<List<String>> res = new ArrayList<>();
        this.max = n;
        this.arr = new int[n];
        check(0);
        print();
        return res;

    }

    public static void main(String[] args) {
        new _51_solveNQueens().solveNQueens(8);
    }

}
